/**
 * 通用js方法封装处理
 * Copyright (c) 2019 ruoyi
 */
import { getToken } from "@/utils/auth";
import errorCode from "@/utils/errorCode";
import axios from "axios";
import ExcelJS from "exceljs"; // 引入exceljs, 用于生成excel文件
import { PxMessage } from "px-ui";

// 日期格式化
export function parseTime(time, pattern) {
  if (arguments.length === 0 || !time) {
    return null;
  }
  const format = pattern || "{y}-{m}-{d} {h}:{i}:{s}";
  let date;
  if (typeof time === "object") {
    date = time;
  } else {
    if (typeof time === "string" && /^[0-9]+$/.test(time)) {
      time = parseInt(time);
    } else if (typeof time === "string") {
      time = time
        .replace(new RegExp(/-/gm), "/")
        .replace("T", " ")
        .replace(new RegExp(/\.[\d]{3}/gm), "");
    }
    if (typeof time === "number" && time.toString().length === 10) {
      time = time * 1000;
    }
    date = new Date(time);
  }
  const formatObj = {
    y: date.getFullYear(),
    m: date.getMonth() + 1,
    d: date.getDate(),
    h: date.getHours(),
    i: date.getMinutes(),
    s: date.getSeconds(),
    a: date.getDay()
  };
  const time_str = format.replace(/{(y|m|d|h|i|s|a)+}/g, (result, key) => {
    let value = formatObj[key];
    // Note: getDay() returns 0 on Sunday
    if (key === "a") {
      return ["日", "一", "二", "三", "四", "五", "六"][value];
    }
    if (result.length > 0 && value < 10) {
      value = "0" + value;
    }
    return value || 0;
  });
  return time_str;
}

// 表单重置
export function resetForm(refName) {
  if (this.$refs[refName]) {
    this.$refs[refName].resetFields();
  }
}

// 添加日期范围
export function addDateRange(params, dateRange, propNameStart, propNameEnd) {
  let search = params;
  search = typeof search === "object" && search !== null && !Array.isArray(search) ? search : {};
  dateRange = Array.isArray(dateRange) ? dateRange : [];
  if (typeof propNameStart === "undefined" && typeof propNameEnd === "undefined") {
    search["startTime"] = dateRange[0];
    search["endTime"] = dateRange[1];
  } else {
    search[propNameStart] = dateRange[0];
    search[propNameEnd] = dateRange[1];
  }
  return search;
}

// 回显数据字典
export function selectDictLabel(datas, value) {
  if (value === undefined) {
    return "";
  }
  var actions = [];
  Object.keys(datas).some(key => {
    if (datas[key].value == "" + value) {
      actions.push(datas[key].label);
      return true;
    }
  });
  if (actions.length === 0) {
    actions.push(value);
  }
  return actions.join("");
}

// 回显数据字典（字符串数组）
export function selectDictLabels(datas, value, separator) {
  if (value === undefined || value.length === 0) {
    return "";
  }
  if (Array.isArray(value)) {
    value = value.join(",");
  }
  var actions = [];
  var currentSeparator = undefined === separator ? "," : separator;
  var temp = value.split(currentSeparator);
  Object.keys(value.split(currentSeparator)).some(val => {
    var match = false;
    Object.keys(datas).some(key => {
      if (datas[key].value == "" + temp[val]) {
        actions.push(datas[key].label + currentSeparator);
        match = true;
      }
    });
    if (!match) {
      actions.push(temp[val] + currentSeparator);
    }
  });
  return actions.join("").substring(0, actions.join("").length - 1);
}

// 字符串格式化(%s )
export function sprintf(str) {
  var args = arguments,
    flag = true,
    i = 1;
  str = str.replace(/%s/g, function () {
    var arg = args[i++];
    if (typeof arg === "undefined") {
      flag = false;
      return "";
    }
    return arg;
  });
  return flag ? str : "";
}

// 转换字符串，undefined,null等转化为""
export function parseStrEmpty(str) {
  if (!str || str == "undefined" || str == "null") {
    return "";
  }
  return str;
}

// 数据合并
export function mergeRecursive(source, target) {
  for (var p in target) {
    try {
      if (target[p].constructor == Object) {
        source[p] = mergeRecursive(source[p], target[p]);
      } else {
        source[p] = target[p];
      }
    } catch (e) {
      source[p] = target[p];
    }
  }
  return source;
}

/**
 * 构造树型结构数据
 * @param {*} data 数据源
 * @param {*} id id字段 默认 'id'
 * @param {*} parentId 父节点字段 默认 'parentId'
 * @param {*} children 孩子节点字段 默认 'children'
 */
export function handleTree(data, id, parentId, children) {
  let config = {
    id: id || "id",
    parentId: parentId || "parentId",
    childrenList: children || "children"
  };

  var childrenListMap = {};
  var nodeIds = {};
  var tree = [];

  for (let d of data) {
    let parentId = d[config.parentId];
    if (childrenListMap[parentId] == null) {
      childrenListMap[parentId] = [];
    }
    nodeIds[d[config.id]] = d;
    childrenListMap[parentId].push(d);
  }

  for (let d of data) {
    let parentId = d[config.parentId];
    if (nodeIds[parentId] == null) {
      tree.push(d);
    }
  }

  for (let t of tree) {
    adaptToChildrenList(t);
  }

  function adaptToChildrenList(o) {
    if (childrenListMap[o[config.id]] !== null) {
      o[config.childrenList] = childrenListMap[o[config.id]];
    }
    if (o[config.childrenList]) {
      for (let c of o[config.childrenList]) {
        adaptToChildrenList(c);
      }
    }
  }
  return tree;
}

/**
 * 参数处理
 * @param {*} params  参数
 */
export function tansParams(params) {
  let result = "";
  for (const propName of Object.keys(params)) {
    const value = params[propName];
    var part = encodeURIComponent(propName) + "=";
    if (value !== null && value !== "" && typeof value !== "undefined") {
      if (typeof value === "object") {
        for (const key of Object.keys(value)) {
          if (value[key] !== null && value[key] !== "" && typeof value[key] !== "undefined") {
            let params = propName + "[" + key + "]";
            var subPart = encodeURIComponent(params) + "=";
            result += subPart + encodeURIComponent(value[key]) + "&";
          }
        }
      } else {
        result += part + encodeURIComponent(value) + "&";
      }
    }
  }
  return result;
}

// 返回项目路径
export function getNormalPath(p) {
  if (p.length === 0 || !p || p == "undefined") {
    return p;
  }
  let res = p.replace("//", "/");
  if (res[res.length - 1] === "/") {
    return res.slice(0, res.length - 1);
  }
  return res;
}

// 验证是否为blob格式
export function blobValidate(data) {
  return data.type !== "application/json";
}

// 通用导出方法
export function download(method, url, data, filename) {
  const { VITE_API_URL: proxy_url = "/" } = import.meta.env;
  // downloadLoadingInstance = ElLoading.service({ text: "正在下载数据，请稍候", background: "rgba(0, 0, 0, 0.7)", })
  return axios({
    method,
    url: `${proxy_url}${url}`,
    data,
    headers: { Authorization: "Bearer " + getToken().accessToken || "", facCode: localStorage.getItem("facCode") || "" },
    responseType: "blob"
  })
    .then(async res => {
      const isBlob = blobValidate(res.data);
      if (isBlob) {
        const blob = res.data;
        const reader = new FileReader();
        reader.readAsDataURL(blob);
        reader.onload = function (e) {
          const a = document.createElement("a");
          let fileName = filename;
          if (res.headers["content-disposition"]) {
            const fileNameEncode = res.headers["content-disposition"].split("filename=")[1].split(";")[0];
            const name = decodeURI(fileNameEncode, "utf-8").replaceAll('"', "");
            if (name) {
              fileName = name;
            }
          }
          a.download = decodeURIComponent(fileName);
          a.href = e.target.result;
          document.body.appendChild(a);
          a.click();
          document.body.removeChild(a);
        };
      } else {
        const resText = await res.data;
        const rspObj = JSON.parse(resText);
        const errMsg = errorCode[rspObj.code] || rspObj.msg || errorCode["default"];
        PxMessage({
          type: "error",
          message: errMsg
        });
      }
      // downloadLoadingInstance.close();
    })
    .catch(r => {
      console.error(r);
      PxMessage({
        type: "error",
        message: "下载文件出现错误，请联系管理员！"
      });
      // downloadLoadingInstance.close();
    });
}
export const downloadDataByExcel = async (tableColumns, tabelData, fileName = "数据导出.xlsx") => {
  if (!tableColumns || !tabelData) return;
  const data = tabelData.map(item => {
    const obj = {};
    tableColumns.forEach(element => {
      obj[element.label] = item[element.prop];
    });
    return obj;
  });
  try {
    // 创建工作簿
    const workbook = new ExcelJS.Workbook();
    workbook.creator = "ExcelJS";
    workbook.created = new Date();

    // 创建工作表
    const worksheet = workbook.addWorksheet("Sheet1");

    // 检查是二维数组还是对象数组
    if (Array.isArray(data[0])) {
      // 二维数组
      data.forEach((row, rowIndex) => {
        worksheet.addRow(row);

        // 设置表头样式
        if (rowIndex === 0) {
          const headerRow = worksheet.getRow(1);
          headerRow.eachCell(cell => {
            cell.fill = {
              type: "pattern",
              pattern: "solid",
              fgColor: { argb: "FFE7EEFB" }
            };
            cell.font = {
              bold: true,
              color: { argb: "FF000000" }
            };
          });
        }
      });
    } else if (typeof data[0] === "object") {
      // 对象数组
      // 添加表头
      const headers = Object.keys(data[0]);
      const headerRow = worksheet.addRow(headers);

      // 设置表头样式
      headerRow.eachCell(cell => {
        cell.fill = {
          type: "pattern",
          pattern: "solid",
          fgColor: { argb: "FFE7EEFB" }
        };
        cell.font = {
          bold: true,
          color: { argb: "FF000000" }
        };
      });

      // 添加数据行
      data.forEach(item => {
        const rowData = headers.map(header => item[header]);
        worksheet.addRow(rowData);
      });
    } else {
      PxMessage({
        type: "error",
        message: "不支持的数据格式"
      });
      return;
    }
    // 自动调整列宽
    worksheet.columns.forEach(column => {
      let maxLength = 0;
      column.eachCell({ includeEmpty: true }, cell => {
        const cellLength = cell.value ? cell.value.toString().length : 10;
        if (cellLength > maxLength) {
          maxLength = cellLength;
        }
      });
      column.width = maxLength * 1.6 + 4;
    });

    // 生成Excel文件并下载
    const buffer = await workbook.xlsx.writeBuffer();
    const blob = new Blob([buffer], {
      type: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
    });

    const url = URL.createObjectURL(blob);
    const a = document.createElement("a");
    a.href = url;
    a.download = fileName;
    document.body.appendChild(a);
    a.click();

    // 清理
    setTimeout(() => {
      document.body.removeChild(a);
      URL.revokeObjectURL(url);
      PxMessage.success("Excel文件下载成功!");
    }, 100);
  } catch (error) {
    PxMessage({
      type: "error",
      message: `生成Excel失败: ${error.message}`
    });
    console.error(error);
  }
};
// 通用获取员工号+用户名方法
export function getEmployeeIdAndName(deptNameList?: Array<string>) {
  const employeeIdAndName = [];
  return axios({
    method: "post",
    url: "/ifp/account/api/v1/user/listPage",
    data: {
      page: 1,
      pageSize: 9999,
      deptNameList,
      status: 0
    },
    headers: {
      "Content-Type": "application/json",
      Authorization: "Bearer " + getToken().accessToken || "",
      facCode: localStorage.getItem("facCode") || ""
    }
  })
    .then(result => {
      if (result.data.code === "0000") {
        result?.data?.data?.data?.forEach(item => {
          employeeIdAndName.push({ ...item, value: item.userId, label: item.username + "-" + item.realName });
        });
        return employeeIdAndName;
      } else {
        PxMessage.error(result.data.msg);
        return [];
      }
    })
    .catch(error => {
      PxMessage.error(`${error}`);
      return [];
    });
}

export function getEmployeeIdAndNameByClassCode(deptNameList, valueField?) {
  const employeeIdAndName = [];
  return axios({
    method: "post",
    url: "/ifp/account/api/v1/user/listPage",
    data: {
      page: 1,
      pageSize: 9999,
      deptNameList,
      status: 0
    },
    headers: {
      "Content-Type": "application/json",
      Authorization: "Bearer " + getToken().accessToken || "",
      facCode: localStorage.getItem("facCode") || ""
    }
  })
    .then(result => {
      if (result.data.success) {
        result?.data?.data?.data.forEach(item => {
          if (valueField) {
            employeeIdAndName.push({ value: item[valueField], label: item.username + "-" + item.realName, phone: item.phone, userId: item.userId });
          } else {
            employeeIdAndName.push({ value: item.userId, label: item.username + "-" + item.realName, phone: item.phone, userId: item.userId });
          }
        });
        return employeeIdAndName;
      } else {
        PxMessage.error(result.data.msg);
        return [];
      }
    })
    .catch(error => {
      PxMessage.error(`${error}`);
      return [];
    });
}
// 时间戳转换天时分秒
export function formatDuration(ms) {
  if (ms === 0) return 0;
  let result = "";
  if (ms < 0) {
    ms = -ms;
    result = "-";
  }
  let seconds = Math.floor(ms / 1000);
  let minutes = Math.floor(seconds / 60);
  let hours = Math.floor(minutes / 60);
  let days = Math.floor(hours / 24);

  seconds = seconds % 60;
  minutes = minutes % 60;
  hours = hours % 24;

  if (days > 0) {
    result += days + "d ";
  }
  if (hours > 0 || (days > 0 && (minutes > 0 || seconds > 0))) {
    result += hours + "h ";
  }
  if (minutes > 0 || ((days > 0 || hours > 0) && seconds > 0)) {
    result += minutes + "m ";
  }
  if (seconds > 0) {
    result += seconds + "s";
  }

  return result.trim();
}

// 权限判断表格中某列显示与否
export function showRowText(val) {
  const permissionList = localStorage.getItem("permissionList");
  const pl = permissionList.split(",");
  if (permissionList === "**" || (pl?.length && pl?.includes(val))) {
    return true;
  } else {
    return false;
  }
}
